package com.sniper.springmvc.shiro;

import com.sniper.springmvc.model.AdminGroup;
import com.sniper.springmvc.model.AdminRight;
import com.sniper.springmvc.model.AdminUser;
import com.sniper.springmvc.mybatis.service.impl.AdminUserService;
import com.sniper.springmvc.mybatis.service.impl.LoginLogService;
import com.sniper.springmvc.shiro.filter.KickoutSessionControlFilter;
import com.sniper.springmvc.shiro.oauth.Constants;
import com.sniper.springmvc.utils.HttpRequestUtils;
import org.apache.commons.lang3.StringUtils;
import org.apache.commons.lang3.builder.ReflectionToStringBuilder;
import org.apache.commons.lang3.builder.ToStringStyle;
import org.apache.shiro.SecurityUtils;
import org.apache.shiro.authc.*;
import org.apache.shiro.authz.AuthorizationException;
import org.apache.shiro.authz.AuthorizationInfo;
import org.apache.shiro.authz.SimpleAuthorizationInfo;
import org.apache.shiro.realm.AuthorizingRealm;
import org.apache.shiro.session.Session;
import org.apache.shiro.subject.PrincipalCollection;
import org.apache.shiro.subject.Subject;
import org.springframework.beans.factory.annotation.Autowired;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import java.util.Date;
import java.util.HashSet;
import java.util.Set;

public class MyShiro extends AuthorizingRealm {

    @Resource
    private AdminUserService userService;
    @Autowired
    HttpServletRequest request;

    @Resource
    LoginLogService loginLogService;

    /**
     * 为当前登录的Subject授予角色和权限
     * <p>
     * 经测试:本例中该方法的调用时机为需授权资源被访问时
     * 经测试:并且每次访问需授权资源时都会执行该方法中的逻辑,这表明本例中默认并未启用AuthorizationCache
     * 个人感觉若使用了Spring3
     * .1开始提供的ConcurrentMapCache支持,则可灵活决定是否启用AuthorizationCache
     * 比如说这里从数据库获取权限信息时,先去访问Spring3.1提供的缓存,而不使用Shior提供的AuthorizationCache
     */
    @Override
    protected AuthorizationInfo doGetAuthorizationInfo(PrincipalCollection principals) {

        // 获取当前登录的用户名,等价于(String)principals.fromRealm(this.getName()).iterator().next()
        String currentUsername = (String) super
                .getAvailablePrincipal(principals);
        Set<String> roleList = new HashSet<>();
        Set<String> permissionList = new HashSet<>();
        // //从数据库中获取当前登录用户的详细信息
        AdminUser user = userService.getUserAndGroupAndRight(currentUsername);
        if (null != user) {
            // 实体类User中包含有用户角色的实体类信息
            if (null != user.getAdminGroup() && user.getAdminGroup().size() > 0) {
                // 获取当前登录用户的角色
                for (AdminGroup role : user.getAdminGroup()) {
                    roleList.add(role.getValue());
                    // 实体类Role中包含有角色权限的实体类信息
                    if (null != role.getAdminRight()
                            && role.getAdminRight().size() > 0) {
                        // 获取权限
                        for (AdminRight pmss : role.getAdminRight()) {
                            if (!StringUtils.isEmpty(pmss.getPermission())) {
                                permissionList.add(pmss.getPermission());
                            }
                        }
                    }
                }
            }
        } else {
            throw new AuthorizationException();
        }
        // //为当前用户设置角色和权限
        SimpleAuthorizationInfo info = new SimpleAuthorizationInfo();
        info.addRoles(roleList);
        info.addStringPermissions(permissionList);

        // 若该方法什么都不做直接返回null的话,就会导致任何用户访问/admin/listUser.jsp时都会自动跳转到unauthorizedUrl指定的地址
        // 详见applicationContext.xml中的<bean id="shiroFilter">的配置
        return info;
    }

    /**
     * 验证当前登录的Subject
     * <p>
     * 经测试:本例中该方法的调用时机为LoginController.login()方法中执行Subject.login()时
     * doGetAuthorizationInfo
     */
    @Override
    protected AuthenticationInfo doGetAuthenticationInfo(AuthenticationToken authcToken) throws AuthenticationException {
        // 获取基于用户名和密码的令牌
        // 实际上这个authcToken是从LoginController里面currentUser.login(token)传过来的
        // 两个token的引用都是一样的
        UsernamePasswordToken token = (UsernamePasswordToken) authcToken;
        System.out.println("验证当前Subject时获取到token为" + ReflectionToStringBuilder.toString(token, ToStringStyle.MULTI_LINE_STYLE));
        AdminUser user;
        try {
            user = userService.getUserAndGroupAndRight(token.getUsername());
        } catch (Exception e) {
            e.printStackTrace();
            // 没找到帐号
            throw new UnknownAccountException();
        }

        if (user == null) {
            throw new UnknownAccountException();// 没找到帐号
        }
        if (Boolean.TRUE.equals(user.getLocked())) {
            throw new LockedAccountException(); // 帐号锁定
        }

        if (Boolean.FALSE.equals(user.getEnabled())) {
            throw new LockedAccountException(); // 帐号未启用
        }

        AuthenticationInfo authcInfo = new SimpleAuthenticationInfo(user.getName(), user.getPassword(), user.getNickName());
        //记录用户登录时间
        this.setSession(KickoutSessionControlFilter.OUT_TIME, new Date());
        this.setSession(Constants.CURRENT_USER.toString(), user);

        //由于ip不能再类里面获取所以只能放在这里了
        String ip = HttpRequestUtils.getRealIP(request);
        this.setSession(KickoutSessionControlFilter.OUT_IP, ip);
        return authcInfo;

        // 此处无需比对,比对的逻辑Shiro会做,我们只需返回一个和令牌相关的正确的验证信息
        // 说白了就是第一个参数填登录用户名,第二个参数填合法的登录密码(可以是从数据库中取到的,本例中为了演示就硬编码了)
        // 这样一来,在随后的登录页面上就只有这里指定的用户和密码才能通过验证
        // 没有返回登录用户名对应的SimpleAuthenticationInfo对象时,就会在 LoginController 中抛出 UnknownAccountException 异常
        // return null;
    }

    /**
     * 将一些数据放到ShiroSession中,以便于其它地方使用
     *
     * <p>
     * 比如Controller,使用时直接用HttpSession.getAttribute(key)就可以取到
     */
    private void setSession(Object key, Object value) {
        Subject currentUser = SecurityUtils.getSubject();
        if (null != currentUser) {
            Session session = currentUser.getSession();
            //System.out.println("Session默认超时时间为[" + session.getTimeout() + "]毫秒");
            session.setAttribute(key, value);
        }
    }
}